package main

import (
	"container/heap"
	"fmt"
)

func main() {
	//在索引 i 处的元素更改其值后，Fix 重新建立堆排序。 更改索引 i 处元素的值，然后调用 Fix 等效于调用 Remove(h, i) 然后调用新值，但成本更低。 复杂度为 O(log n)，其中 n = h.Len()。

	h := &IntHeap{1, 2, 3, 4, 5, 6, 7, 8, 9}

	heap.Init(h)
	heap.Push(h, 10)

	//
	//h.index(9 , 0)
	//heap.Fix(h, 9)

	for h.Len() > 0 {
		fmt.Println(h.Pop())
	}
}

type IntHeap []int

func (h IntHeap) Len() int {

	return len(h)
}

func (h IntHeap) Less(i, j int) bool {
	return h[i] < h[j]
}

func (h IntHeap) Swap(i, j int) {

	h[i], h[j] = h[j], h[i]
}

func (h *IntHeap) Push(x interface{}) {
	*h = append(*h, x.(int))
}

func (h *IntHeap) Pop() interface{} {
	old := *h
	n := len(old)
	x := old[n-1]
	*h = old[0 : n-1]
	return x
}

func (h *IntHeap) index(i, v int) {
	temp := *h
	temp[i] = v
	*h = temp
}
